tensor([[0.4400, 0.4400, 0.4400, 0.4400, 0.4400],
        [0.5600, 0.5600, 0.5600, 0.5600, 0.5600]], grad_fn=<RoundBackward1>)